# -*- coding:utf-8 -*-

from __future__ import unicode_literals

from datetime import date

from flask import render_template, request, redirect, url_for, session, flash

from login_manager import login_required
from modelos import Professor, Disciplina, Aluno, Atividade

from __init__ import app

#cadastrar uma atividade
@app.route("/form_add_atividade/<int:id>", methods=['GET', 'POST'])
@login_required
def form_add_atividade(id):

    p = Professor.get(Professor.id == session["user_id"])
    print "id="+`id`
    disciplina = Disciplina.get(Disciplina.id == id, Disciplina.professor == p)

    if request.method == 'POST':
    	a = Atividade.create(nome=request.form["nome"],
    							tipo=request.form["tipo"],
                                disciplina=disciplina)
    	return redirect(url_for("ver_disciplina", id=disciplina.id))
    else:
		return render_template("form_add_atividade.html",
			disciplina = disciplina)

#remove uma atividade
@app.route("/remover_atividade/<int:id>")
@login_required
def remover_atividade(id):
    a = Atividade.get(Atividade.id == id)
    p = Professor.get(Professor.id == session["user_id"])
    d = Disciplina.get(Disciplina.id == a.disciplina, Disciplina.professor==p)
    a.delete_instance()

    return redirect(url_for("ver_disciplina", id=d.id))